CONFIG = 'PROD'

BATCH_SIZE = 32
NUM_CLASSES = 10
SAVE_DIR = 'weights/'
IMAGE_SIZE = (32, 32)
CHANNELS = 3
DOMAINS = [
    'mnist',
    'svhn'
]
DOMAINS_IGNORE_LABELS = [
    'mnist'
]

EPOCHS = 500
SUBSET = 10000


if CONFIG == 'DEBUG':
    EPOCHS = 40
    SUBSET = 100
